import sys, os
sys.path.insert(0, sys.path[0]+"/../")
import copy
from modules import loadckpts, BuildSegmentor
from myutil.mixconfig import SEGMENTOR_CFG
import torch

if __name__ == "__main__":
    segmentor = BuildSegmentor(segmentor_cfg=copy.deepcopy(SEGMENTOR_CFG), mode='TEST')
    print(segmentor)
    ckpt = loadckpts("fcn_resnet50os16_voc/epoch_3.pth")
    torch.save(ckpt["model"], "a.pt")
